# --*-- coding=utf-8 --*--
"""
对xlrd模块常用方法的初步封装
参考网页：https://www.cnblogs.com/insane-Mr-Li/p/9092619.html

1、常用单元格中的数据类型
♦ 0. empty（空的）,1 string（text）, 2 number, 3 date, 4 boolean, 5 error， 6 blank（空白表格）
2、导入模块
import xlrd

3、打开Excel文件读取数据
#文件名以及路径，如果路径或者文件名有中文给前面加一个r拜师原生字符。
book = xlrd.open_workbook(filename)

4、常用的函数
♦ excel中最重要的方法就是book和sheet的操作

1）获取book中一个工作表
table = data.sheets()[0] #通过索引顺序获取
table = data.sheet_by_index(sheet_indx)) #通过索引顺序获取
table = data.sheet_by_name(sheet_name)#通过名称获取
以上三个函数都会返回一个xlrd.sheet.Sheet()对象
names = book.sheet_names()    #返回book中所有工作表的名字
book.sheet_loaded(sheet_name or indx)   # 检查某个sheet是否导入完毕

2）行的操作

nrows = table.nrows  #获取该sheet中的有效行数
table.row(rowx)  #返回由该行中所有的单元格对象组成的列表
table.row_slice(rowx)  #返回由该列中所有的单元格对象组成的列表
table.row_types(rowx, start_colx=0, end_colx=None)    #返回由该行中所有单元格的数据类型组成的列表
table.row_values(rowx, start_colx=0, end_colx=None)   #返回由该行中所有单元格的数据组成的列表
table.row_len(rowx) #返回该列的有效单元格长度

3）列(colnum)的操作

ncols = table.ncols   #获取列表的有效列数
table.col(colx, start_rowx=0, end_rowx=None)  #返回由该列中所有的单元格对象组成的列表
table.col_slice(colx, start_rowx=0, end_rowx=None)  #返回由该列中所有的单元格对象组成的列表
table.col_types(colx, start_rowx=0, end_rowx=None)    #返回由该列中所有单元格的数据类型组成的列表
table.col_values(colx, start_rowx=0, end_rowx=None)   #返回由该列中所有单元格的数据组成的列表

4）单元格的操作

table.cell(rowx,colx)   #返回单元格对象
table.cell_type(rowx,colx)    #返回单元格中的数据类型
table.cell_value(rowx,colx)   #返回单元格中的数据
table.cell_xf_index(rowx, colx)   # 暂时还没有搞懂

♦单元格：单元格是表格中行与列的交叉部分，它是组成表格的最小单位，可拆分或者合并。单个数据的输入和修改都是在单元格中进行的
work_book常用方法：get_sheets();sheet_by_name();sheet_by_index();
"""
import xlrd
import xlwt


def get_sheet(fp, sheet_name=None):
    """
    获取sheet对象
    :param fp:
    :param sheet_name:
    :return:
    """
    rb = xlrd.open_workbook(fp)
    if sheet_name is None:
        return rb.sheet_by_index(0)
    elif isinstance(sheet_name, int) and sheet_name < len(rb.sheets()):
        return rb.sheet_by_index(sheet_name)
    elif isinstance(sheet_name, str) and sheet_name in rb.sheet_names():
        return rb.sheet_by_name(sheet_name)
    else:
        return None


def sheet_to_list(fp, sheet_name=None):
    """
    :param fp:
    :param sheet_name:
    :return:
    """
    records = []
    sheet = get_sheet(fp, sheet_name)
    if sheet is None:
        return records
    rows = sheet.nrows
    for i in range(rows):
        if i >= 1:
            record = dict(zip(sheet.row_values(0), sheet.row_values(i)))
            records.append(record)
    return records


def excel_to_list(fp):
    """
    :param fp:
    :return:
    """
    rb = xlrd.open_workbook(fp)
    book_list = []
    for sheet in rb.sheets():
        sheet_dict = {}
        records = sheet_to_list(sheet)
        sheet_dict[sheet.name] = records
        book_list.append(sheet_dict)
    return book_list


def list_to_excel(fp, data):
    """

    :param fp:
    :param data:
    """
    wb = xlwt.Workbook(encoding='utf-8')
    sheet_name = "test"
    table = wb.add_sheet(sheet_name, cell_overwrite_ok=True)
    # 使用style
    style = xlwt.XFStyle()  # 初始化样式
    font = xlwt.Font()  # 为样式创建字体
    font.name = 'Times New Roman'
    font.bold = False
    # font.height=
    style.font = font  # 为样式设置字体
    for rowindex, items in enumerate(data):
        for colindex, item in enumerate(items):
            table.write(rowindex, colindex, item, style)
    wb.save(fp)
